#!/usr/bin/env bash

domain_description="TEC is an important information for describing the state of the ionosphere, as predicting ionospheric TEC can analyze the impact of space weather on the ionosphere, and thus analyze the impact of space weather on global navigation satellite systems (GNSS)."

LLM_model=Qwen/Qwen2-0.5B-Instruct

llm_model_root=./LLMs

TEC_Data_Path=./data/TEC_SpaceWeather_Data.csv


accelerate launch --multi_gpu --num_processes=2 --mixed_precision=fp16 train_TimeLLMForTEC_distribution.py \
    --llm_model $LLM_model \
    --llm_model_root $llm_model_root \
    --domain_description "$domain_description" \
    --num_features 6 \
    --seq_len 336 \
    --pred_len 96 \
    --patch_len 6 \
    --patch_stride 3 \
    --d_merge 16 \
    --d_model 32 \
    --n_heads 4 \
    --d_ff 96 \
    --text_prototype_num 1000 \
    --top_k 4 \
    --model_name TimeLLMForTEC_1 \
    --model_save_path ./models \
    --batch_size 40 \
    --epochs 50 \
    --learning_rate 0.001 \
    --num_workers 8 \
    --gamma 0.98 \
    --patience 5 \
    --delta 0.2 \
    --TEC_Data_Path $TEC_Data_Path \
    --start_date 2013/01/01 \
    --end_date 2016/01/01 \
    --train_valid_date 2015/01/01 \
    --sample_stride 48
